package xyz.ibenben.zhongdian.common.util.example.middle;

import xyz.ibenben.zhongdian.common.util.example.constant.ListNode;

/**
 * 给出两个 非空 的链表用来表示两个非负的整数。其中，它们各自的位数是按照 逆序 的方式存储的，并且它们的每个节点只能存储 一位 数字。
 * 如果，我们将这两个数相加起来，则会返回一个新的链表来表示它们的和。
 * 您可以假设除了数字 0 之外，这两个数都不会以 0 开头。
 * <p>
 * 示例：
 * 输入：(2 -> 4 -> 3) + (5 -> 6 -> 4)
 * 输出：7 -> 0 -> 8
 * 原因：342 + 465 = 807
 */
public class AddTwoNumber {
    public static void main(String[] args) {
        AddTwoNumber atn = new AddTwoNumber();
        ListNode l1 = new ListNode(1);
        l1.next = new ListNode(8);
        ListNode l2 = new ListNode(0);
        System.out.println(atn.addTwoNumbers(l1, l2));
        System.out.println(atn.addTwoNumbers2(l1, l2));
    }

    /**
     * 我给出的解决方案
     *
     * @param l1
     * @param l2
     * @return
     */
    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        ListNode result = new ListNode(0);
        ListNode temp = result;
        int bigCount = 0;
        while (l1 != null || l2 != null) {
            int x = (l1 != null) ? l1.val : 0;
            int y = (l2 != null) ? l2.val : 0;
            if ((x + y + bigCount) >= 10) {
                temp.next = new ListNode(x + y + bigCount - 10);
                bigCount = 1;
            } else {
                if (bigCount != 0) {
                    temp.next = new ListNode(x + y + 1);
                    bigCount = 0;
                } else {
                    temp.next = new ListNode(x + y);
                }
            }
            temp = temp.next;
            if (l1 != null) {
                l1 = l1.next;
            }
            if (l2 != null) {
                l2 = l2.next;
            }
        }
        if (bigCount > 0) {
            temp.next = new ListNode(bigCount);
        }
        return result.next;
    }

    /**
     * 官方给出的解决方案
     *
     * @param l1
     * @param l2
     * @return
     */
    public ListNode addTwoNumbers2(ListNode l1, ListNode l2) {
        ListNode dummyHead = new ListNode(0);
        ListNode p = l1, q = l2, curr = dummyHead;
        int carry = 0;
        while (p != null || q != null) {
            int x = (p != null) ? p.val : 0;
            int y = (q != null) ? q.val : 0;
            int sum = carry + x + y;
            carry = sum / 10;
            curr.next = new ListNode(sum % 10);
            curr = curr.next;
            if (p != null) p = p.next;
            if (q != null) q = q.next;
        }
        if (carry > 0) {
            curr.next = new ListNode(carry);
        }
        return dummyHead.next;
    }
}
